/*
 *  Copyright 2009 Advanced Industrial Science and Technology (AIST).
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

/*! \file
 *  \brief
 */
#pragma once

#include <thrust/graph/detail/undirected_out_edge_iterator.hpp>

#include <thrust/detail/device/dereference.h>
#include <thrust/iterator/iterator_traits.h>

namespace thrust {

template <typename EdgeIter, typename VertexDescriptor, typename EdgeDescriptor, typename CalledSpace>
struct undirected_out_edge_iter;

namespace detail {
  
struct undirected_out_edge_iter_friend {
  
  template <typename EdgeIter, typename VertexDescriptor, typename EdgeDescriptor, typename CalledSpace>
  static inline __host__ __device__
  typename device::dereference_result< undirected_out_edge_iter<EdgeIter
                                                                , VertexDescriptor
                                                                , EdgeDescriptor
                                                                , CalledSpace> >::type
  dereference( const undirected_out_edge_iter<EdgeIter,VertexDescriptor,EdgeDescriptor,CalledSpace>& iter)
  {
    return iter.dereference( thrust::called_from_device_tag());
  }  
};

namespace device {
  template<typename EdgeIter, typename VertexDescriptor, typename EdgeDescriptor, typename CalledSpace>
  struct dereference_result< undirected_out_edge_iter<EdgeIter
                                                      , VertexDescriptor
                                                      , EdgeDescriptor
                                                      , CalledSpace>
                             >
  {
  typedef typename thrust::iterator_traits< undirected_out_edge_iter<EdgeIter
                                                                     , VertexDescriptor
                                                                     , EdgeDescriptor
                                                                     , CalledSpace> 
                                            >::reference type;
};

template <typename EdgeIter, typename VertexDescriptor, typename EdgeDescriptor, typename CalledSpace>
inline __host__ __device__
typename dereference_result< undirected_out_edge_iter<EdgeIter
                                                      , VertexDescriptor
                                                      , EdgeDescriptor
                                                      , CalledSpace>
                             >::type
dereference( const undirected_out_edge_iter<EdgeIter,VertexDescriptor,EdgeDescriptor,CalledSpace>& iter)
{
  return undirected_out_edge_iter_friend::dereference( iter);
}

} // end namespace device

} // end namespace detail

} // end namespace thrust
